# -*- coding: utf-8 -*-            
# @Author : HJH
# @Time : 2024/5/15 11:59
# @Use :
import pandas as pd

# 读取第一张表的数据
df1 = pd.read_excel('南航NDC报价数据.xlsx')

# 读取第二张表的数据
df2 = pd.read_excel('航段限制信息240619-240626.xls')

# 合并两张表的数据
merged_df = pd.merge(df1, df2, on=['起飞', '到达'], how='left')

# 选择要保留的字段
selected_columns = ['起飞', '到达', '出发日期', '航班号', '舱位', '销售价', '代理费', '面值(元）']  # 这里替换为你想要保留的字段名

# 保留指定字段
merged_df = merged_df[selected_columns]

# 将合并后的DataFrame转换为字典列表
dict_list = merged_df.to_dict(orient='records')

# 去除指定舱位
dict_list = [item for item in dict_list if item['舱位'] not in ['W', 'S', 'J', 'Y']]

# 去除面值为空的
dict_list = [item for item in dict_list if not pd.isnull(item['面值(元）']) or item['代理费'] > 0]

# 去除不符合用券的
dict_list = [item for item in dict_list if (item['面值(元）'] * 5) <= item['销售价']]

for item in dict_list:
    if item['面值(元）'] == 20:
        item['QN竞价'] = item['面值(元）'] + item['代理费'] + 5.6
    elif item['面值(元）'] == 60:
        item['QN竞价'] = item['面值(元）'] + item['代理费'] + 8.3

    item['备注'] = f"{item['销售价']}价 {item['代理费']}代理+{item['面值(元）']}券"

df = pd.DataFrame(dict_list)

# 保存合并后的数据到Excel文件
df.to_excel('合并后的数据.xlsx', index=False)